package com.atguigu.servlet;

import org.apache.commons.io.IOUtils;
import sun.misc.BASE64Encoder;

import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;

/**
 * 类说明：
 *      文件的下载
 *
 * @author ChenCheng
 * @ClassName Download
 * @create 2023-02-05 15:59
 * @Version 1.0
 *
 *
 *              response.getOutputStream();
 *              servletContext.getResourceAsStream();
 *              servletContext.getMimeType();
 *              response.setContentType();
 */
public class DownloadServlet extends HttpServlet {

    /**
     * 方法说明：
     * @MethodName 文件下载的功能
     * @Description    
     * @Author Chen
     * @EMAIL <a href="2277411582@qq.com"></a>
     * @Date 2023/2/5 16:03  
     */
    @Override
    protected void doGet(
            HttpServletRequest request,
            HttpServletResponse response)
            throws ServletException, IOException {
    // 1.获取要下载的文件名
         String downloadFileName = "2.jpg";
    // 2.读取要下载的文件内容 (通过ServletContext对象可以读取)
        ServletContext servletContext = getServletContext();
    // 3.获取要下载的文件类型
        String mimeType = servletContext.getMimeType("/file/" + downloadFileName);
        System.out.println(mimeType);
    // 4.在回传前，通过响应头告诉客户端返回的数据类型
        response.setContentType(mimeType);
    // 5.还要告诉客户端收到的数据是用于下载使用（还是使用响应头）
    // Content-Disposition响应头，表示收到的数据怎么处理
    // attachment表示附件，表示下载使用
    // filename= 表示指定下载的文件名
    // url编码是把汉字转换成为%xx%xx的格式
        if(request.getHeader("User-Agent").contains("Firefox")){
            // 如果是火狐浏览器使用Base64编码
            response.setHeader("Content-Disposition",
                    "attachment;filename==?UTF-8?B?" + new BASE64Encoder().encode("中国.jpg".getBytes("UTF-8")) + "?=");
        }else {
            // 如果不是火狐,使用URL编码操作
            response.setHeader("Content-Disposition",
                    "attachment;filename=" + URLEncoder.encode("中国.jpg", StandardCharsets.UTF_8.name()));
        }
        /**
         * 斜杠被服务器解析表示地址为http://ip:prot/工程名/  映射 到代码的Web目录
         */
        // 获取输入流
        InputStream resourceAsStream = servletContext.getResourceAsStream("/file/" + downloadFileName);
        // 获取响应的输出流
        OutputStream outputStream = response.getOutputStream();
        IOUtils.copy(resourceAsStream, outputStream);
    }





}
